import os
import shutil
from tqdm import tqdm

# Set image-input path
folder_path = 'F:\\items\\bike-model\\搜集的数据集\\解压后\\数据集\\zzxi'

# Set image-output path
output_folder_path = 'F:\\items\\bike-model\\data-v2（1303）\\images'

# Set annotation-input path
annotation_input_path = 'F:\\items\\bike-model\\搜集的数据集\\解压后\\数据集\\zzxa2'

# Set annotation-output path
annotation_output_path = 'F:\\items\\bike-model\\data-v2（1303）\\annotations'

# 获取源文件夹中的图片文件列表
img_files = [f for f in os.listdir(folder_path) if f.endswith('.jpg')]

leg = 1103

for i, file_name in tqdm(enumerate(img_files)):
    # 构造新的文件名
    new_name = '{:04d}.jpg'.format(i + leg + 1)

    # 重命名并复制文件
    shutil.copy(os.path.join(folder_path, file_name), os.path.join(output_folder_path, new_name))

    # 同步处理xml文件
    new_name2 = '{:04d}.xml'.format(i + leg + 1)
    file_name = file_name[:-3] + 'xml'
    shutil.copy(os.path.join(annotation_input_path, file_name), os.path.join(annotation_output_path, new_name2))
